from PIL import Image

# 图像尺寸
image_size = (200, 200)
# 障碍物尺寸
obstacle_size = (50, 50)

# 创建一个白色背景的图像
image = Image.new("L", image_size, 255)  # "L" 模式表示灰度图像，255表示白色

# 绘制黑色障碍物
obstacle_top_left = ((image_size[0] - obstacle_size[0]) // 2, (image_size[1] - obstacle_size[1]) // 2)
for x in range(obstacle_top_left[0], obstacle_top_left[0] + obstacle_size[0]):
    for y in range(obstacle_top_left[1], obstacle_top_left[1] + obstacle_size[1]):
        image.putpixel((x, y), 0)  # 0 表示黑色

# 保存为pgm和png文件
image.save("output.pgm")
image.save("output.png")

print("图像已保存为output.pgm和output.png")
